from langchain_openai import OpenAIEmbeddings
import __init__

e_model = OpenAIEmbeddings()
embeddings = e_model.embed_documents(
    [
        "你好",
        "你好啊",
        "你叫什么名字？",
        "我叫王大锤",
        "很高兴认识你大锤",
    ]
)
print(embeddings)


embedded_query = e_model.embed_query("这段对话中提到了什么名字？")
print(embedded_query[:5])